#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/wait.h>

int main() {
    while(1) {
        printf("ysq> ");

        char line[1024];
        fgets(line, 1024, stdin);
        char *p = strchr(line, '\n');  // fgets会读入换行，需要把line末尾的换行改为'\0'
        *p = '\0';

        int n = 0;
        char *args[64];
        char *delim = " ";
        args[n ++] = strtok(line, delim);
        while((p = strtok(NULL, delim)))
            args[n ++] = p;
        args[n ++] = NULL;

        // 父进程内置命名处理
        if(!strcmp(args[0], "exit")) {
            exit(0);
        } else if(!strcmp(args[0], "pwd")) {
            char buf[1024];
            getcwd(buf, 1024);
            printf("%s\n", buf);
        } else if(!strcmp(args[0], "cd")) {
            char buf[1024];
            getcwd(buf, 1024);
            strcat(buf, "/");
            strcat(buf, args[1]);
            chdir(buf);
        } else {  // 子进程处理
            pid_t pid = fork();
            if(pid == 0) {
                execvp(args[0], args);
            }
            wait(NULL);
        }
    }
    return 0;
}
